package com.hanxiaozhang.aspect;

import com.alibaba.fastjson.JSON;
import com.hanxiaozhang.util.SensitiveUtil;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;

/**
 * 功能描述: <br>
 * 〈所有controller切面〉
 *
 * @Author:hanxiaozhang
 * @Date: 2021/3/19
 */
@Slf4j
@Aspect
@Component
public class WebLogAspect {

    

    /**
     * 环境
     */
    @Value("${hanxiaozhang.log.prefix:pro}")
    private String prefix;

    /**
     * 最大长度
     */
    @Value("${hanxiaozhang.log.max.length:2000}")
    private int maxLength;


    /**
     * 是否启动标识
     */
    @Value("${hanxiaozhang.log.sensitive.enable:true}")
    private boolean sensitiveEnable;


    /**
     * 提供一个方法声明，定义切面的切入点
     * 指示符类型：
     * i.within 类型签名表达式
     * ii.execution 方法签名表达式
     * ..：代表所有子目录，最后括号里的两个..代表所有参数
     * iii.bean(*Service) bean表达式
     * iv.@annotation() 注解
     */
    @Pointcut("execution( * com.hanxiaozhang..controller.*.*(..))")
    public void logPointCut() {
    }


    /**
     * “通知”在实际方法调用之前执行
     *
     * @param joinPoint
     */
    @Before("logPointCut()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        // 接收到请求，记录请求内容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
        // 记录下请求内容
        log.info("请求地址:[{}]", request.getRequestURL().toString());
        log.info("HTTP METHOD:[{}]", request.getMethod());
        log.info("CLASS_METHOD:[{}]", joinPoint.getSignature().getDeclaringTypeName() + "."
                + joinPoint.getSignature().getName());
        log.info("参数:[{}]", Arrays.toString(joinPoint.getArgs()));
    }


    /**
     * “通知”在实际方法调用之后执行，可以获取实现方法的返回值，
     * 如果从一个“通知”方法中抛出一个异常，则不会执行该“通知”
     * <p>
     * Tips:returning的值和doAfterReturning的形式参数名一致
     *
     * @param ret
     */
    @AfterReturning(returning = "ret", pointcut = "logPointCut()")
    public void doAfterReturning(Object ret) throws Throwable {
        // 处理完请求，返回内容
        log.info("返回值:[{}]", objectToString(ret));
    }


    /**
     * 环绕通知,
     * Tips：它可以修改实际方法返回的值和调用@AfterReturning返回的值
     *
     * @param pjp
     * @return
     * @throws Throwable
     */
    @Around("logPointCut()")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        long startTime = System.currentTimeMillis();
        Object ob = pjp.proceed();
        log.info("耗时:[{}]", (System.currentTimeMillis() - startTime));
        return ob;
    }


    /**
     * 将对象转为字符串作为log输出
     *
     * @param obj
     * @return 字符串
     */
    private String objectToString(Object obj) {
        // 对象为空
        if (obj == null) {
            return "";
        }
        // 对象为枚举类
        if (obj instanceof Enum) {
            return obj.toString();
        }
        try {
            String json = null;
            if (sensitiveEnable) {
                json = SensitiveUtil.toJsonString(obj);
            } else {
                json = JSON.toJSONString(obj);
            }
            if (json.length() > maxLength) {
                return json.substring(0, maxLength) + "...";
            }
            return json;
        } catch (Exception e) {
            log.warn(e.getMessage());
            return obj.toString();
        }
    }
}
